package com.xj.algorithm.leetcode;

import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

/**
 * 存在重复元素II
 */
public class L219_数组中存在位置相差大于K的重复元素 {

    /**
     * 给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的绝对值最大为 k。
     *
     * 示例 1:
     *
     * 输入: nums = [1,2,3,1], k = 3
     * 输出: true
     * 示例 2:
     *
     * 输入: nums = [1,0,1,1], k = 1
     * 输出: true
     * 示例 3:
     *
     * 输入: nums = [1,2,3,1,2,3], k = 2
     * 输出: false
     */

    public static void main(String[] args) {
        int[] nums=new int[]{1,2,3,1};
        int k=3;
        System.out.println(containsNearbyDuplicateOffice(nums,k));

        int[] nums2=new int[]{1,0,1,1};
        int k2=1;
        System.out.println(containsNearbyDuplicateOffice(nums2,k2));

        int[] nums3=new int[]{1,2,3,1,2,3};
        int k3=2;
        System.out.println(containsNearbyDuplicateOffice(nums3,k3));
    }

    //自己的解法：双重循环
    public static boolean containsNearbyDuplicate(int[] nums, int k) {
        int length=nums.length;
        for (int i=0;i<length;i++){
            int count=i+k>length-1?length-1:i+k;
            for (int j=i+1;j<=count;j++){
               if(nums[i]==nums[j]){
                   return true;
               }
            }
        }
        return false;
    }

    //官方解法：hashset的滑动窗口
    public static boolean containsNearbyDuplicateOffice(int[] nums, int k) {
        HashSet<Integer> set = new HashSet<>();
        for(int i = 0; i < nums.length; i++) {
            if(set.contains(nums[i])) {
                return true;
            }
            set.add(nums[i]);
            if(set.size() > k) {
                set.remove(nums[i - k]);
            }
        }
        return false;
    }
}